import torch
print(f"PyTorch CUDA 可用: {torch.cuda.is_available()}")  # 应输出 True
print(f"当前 GPU 设备: {torch.cuda.get_device_name(0)}")   # 应显示 GPU 型号（如 NVIDIA RTX 3090）